import numpy as np

def change(lst):
    '''
    利用ufunc，实现将lst中下标为2和4的元素替换成相应的符号，正的值替换成1，负值替换成-1。用一行代码实现。
    '''

def test_change():
    np.random.seed(44)
    arr = np.random.randint(-4, high = 4, size = 7)
    sign2 = np.sign(arr[2])
    sign4 = np.sign(arr[4])
    change(arr)
    assert sign2 == arr[2]
    assert sign4 == arr[4]
